#ifndef __GOFGAME_DUNGEONKEEPER_UI_CHOICE_UNIT_VIEW_H__
#define __GOFGAME_DUNGEONKEEPER_UI_CHOICE_UNIT_VIEW_H__

#include "gofgame/dungeonkeeper/ui/MessageViewDialog.h"
#include "gofgame/common/ui/ChoiceView.h"

namespace gofgame {
	namespace dungeonkeeper {
		namespace ui {
			class ChoiceUnitDialog : public MessageViewDialog {
			public:
				ChoiceUnitDialog();
				virtual bool init() _override;
				virtual void onDialogClosed(DialogClosedEventArgs &e);
			protected:
				virtual Ref<Component> createMainComponent() _override;

			private:
				class UnitTemplate : public Template
				{
				public:
					virtual Ref<Component> createComponent(CCObject *dataItem) _override;
				};
			};
		}
	}
}

#endif //__GOFGAME_DUNGEONKEEPER_UI_CHOICE_UNIT_VIEW_H__
